package io.renren.modules.medicine.controller;

import io.renren.common.annotation.LogOperation;
import io.renren.common.constant.Constant;
import io.renren.common.page.PageData;
import io.renren.common.utils.ExcelUtils;
import io.renren.common.utils.NfsUtil;
import io.renren.common.utils.Result;
import io.renren.common.validator.AssertUtils;
import io.renren.common.validator.ValidatorUtils;
import io.renren.common.validator.group.AddGroup;
import io.renren.common.validator.group.DefaultGroup;
import io.renren.common.validator.group.UpdateGroup;
import io.renren.modules.medicine.dto.TDatasetDTO;
import io.renren.modules.medicine.dto.TaskProteinSequenceDesignDTO;
import io.renren.modules.medicine.excel.TaskProteinSequenceDesignExcel;
import io.renren.modules.medicine.service.TDatasetService;
import io.renren.modules.medicine.service.TaskProteinSequenceDesignService;
import io.swagger.annotations.Api;
import io.swagger.annotations.ApiImplicitParam;
import io.swagger.annotations.ApiImplicitParams;
import io.swagger.annotations.ApiOperation;
import org.apache.shiro.authz.annotation.RequiresPermissions;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.web.bind.annotation.*;
import org.springframework.web.multipart.MultipartFile;
import springfox.documentation.annotations.ApiIgnore;

import javax.servlet.http.HttpServletResponse;
import java.io.ByteArrayInputStream;
import java.io.IOException;
import java.io.InputStream;
import java.util.List;
import java.util.Map;


/**
 * 蛋白质序列设计
 *
 * @author sheliming shelimingming@qq.com
 * @since 1.0.0 2023-02-28
 */
@RestController
@RequestMapping("medicine/taskproteinsequencedesign")
@Api(tags="蛋白质序列设计")
public class TaskProteinSequenceDesignController {
    @Autowired
    private TaskProteinSequenceDesignService taskProteinSequenceDesignService;
    @Autowired
    private TDatasetService tDatasetService;

    @PostMapping(value = "/putTaskAsyn", headers = "content-type=multipart/form-data")
    @ApiOperation("异步创建任务")
    @RequiresPermissions("medicine:taskproteinsequencedesign:save")
    public Result putTaskAsyn(MultipartFile pdb) throws IOException {
        InputStream inputStream = pdb.getInputStream();
        String originalFilename = pdb.getOriginalFilename();

        String filePath = NfsUtil.uploadFile(originalFilename, inputStream);

        System.out.println(filePath);

        taskProteinSequenceDesignService.putTaskAsyn(filePath);

        return new Result();
    }

    @PostMapping(value = "/putTaskDatasetAsyn", headers = "content-type=multipart/form-data")
    @ApiOperation("异步创建任务Dataset")
    @RequiresPermissions("medicine:taskproteinsequencedesign:save")
    public Result putTaskAsyn(Long datasetId) throws IOException {
        TDatasetDTO tDatasetDTO = tDatasetService.get(datasetId);
        ByteArrayInputStream byteArrayInputStream = new ByteArrayInputStream(tDatasetDTO.getDataset().getBytes());
        String filePath = NfsUtil.uploadFile("string", byteArrayInputStream);

        System.out.println(filePath);

        taskProteinSequenceDesignService.putTaskAsyn(filePath);

        return new Result();
    }

    @PostMapping(value = "/putTaskStringAsyn", headers = "content-type=multipart/form-data")
    @ApiOperation("异步创建任务String")
    @RequiresPermissions("medicine:taskproteinsequencedesign:save")
    public Result putTaskAsyn(String pdb) throws IOException {
        ByteArrayInputStream byteArrayInputStream = new ByteArrayInputStream(new String(pdb).getBytes());
        String filePath = NfsUtil.uploadFile("string", byteArrayInputStream);

        System.out.println(filePath);

        taskProteinSequenceDesignService.putTaskAsyn(filePath);

        return new Result();
    }

    @PostMapping(value = "/putTask", headers = "content-type=multipart/form-data")
    @ApiOperation("创建任务")
    @RequiresPermissions("medicine:taskproteinsequencedesign:save")
    public Result putTask(MultipartFile pdb) throws IOException {
        InputStream inputStream = pdb.getInputStream();
        String originalFilename = pdb.getOriginalFilename();

        String filePath = NfsUtil.uploadFile(originalFilename, inputStream);

        System.out.println(filePath);

        String result = taskProteinSequenceDesignService.putTask(filePath);

        return new Result<String>().ok(result);
    }

    @PostMapping(value = "/putTaskDataset", headers = "content-type=multipart/form-data")
    @ApiOperation("创建任务Dataset")
    @RequiresPermissions("medicine:taskproteinsequencedesign:save")
    public Result putTask(Long datasetId) throws IOException {
        TDatasetDTO tDatasetDTO = tDatasetService.get(datasetId);
        ByteArrayInputStream byteArrayInputStream = new ByteArrayInputStream(tDatasetDTO.getDataset().getBytes());
        String filePath = NfsUtil.uploadFile("string", byteArrayInputStream);

        System.out.println(filePath);

        String result = taskProteinSequenceDesignService.putTask(filePath);

        return new Result<String>().ok(result);
    }

    @PostMapping(value = "/putTaskString", headers = "content-type=multipart/form-data")
    @ApiOperation("创建任务String")
    @RequiresPermissions("medicine:taskproteinsequencedesign:save")
    public Result putTask(String pdb) throws IOException {
        ByteArrayInputStream byteArrayInputStream = new ByteArrayInputStream(new String(pdb).getBytes());
        String filePath = NfsUtil.uploadFile("string", byteArrayInputStream);

        System.out.println(filePath);

        String result = taskProteinSequenceDesignService.putTask(filePath);

        return new Result<String>().ok(result);
    }

    @GetMapping("page")
    @ApiOperation("分页")
    @ApiImplicitParams({
        @ApiImplicitParam(name = Constant.PAGE, value = "当前页码，从1开始", paramType = "query", required = true, dataType="int") ,
        @ApiImplicitParam(name = Constant.LIMIT, value = "每页显示记录数", paramType = "query",required = true, dataType="int") ,
        @ApiImplicitParam(name = Constant.ORDER_FIELD, value = "排序字段", paramType = "query", dataType="String") ,
        @ApiImplicitParam(name = Constant.ORDER, value = "排序方式，可选值(asc、desc)", paramType = "query", dataType="String")
    })
    @RequiresPermissions("medicine:taskproteinsequencedesign:page")
    public Result<PageData<TaskProteinSequenceDesignDTO>> page(@ApiIgnore @RequestParam Map<String, Object> params){
        PageData<TaskProteinSequenceDesignDTO> page = taskProteinSequenceDesignService.page(params);

        return new Result<PageData<TaskProteinSequenceDesignDTO>>().ok(page);
    }

    @GetMapping("{id}")
    @ApiOperation("信息")
    @RequiresPermissions("medicine:taskproteinsequencedesign:info")
    public Result<TaskProteinSequenceDesignDTO> get(@PathVariable("id") Long id){
        TaskProteinSequenceDesignDTO data = taskProteinSequenceDesignService.get(id);

        return new Result<TaskProteinSequenceDesignDTO>().ok(data);
    }

    @PostMapping
    @ApiOperation("保存")
    @LogOperation("保存")
    @RequiresPermissions("medicine:taskproteinsequencedesign:save")
    public Result save(@RequestBody TaskProteinSequenceDesignDTO dto){
        //效验数据
        ValidatorUtils.validateEntity(dto, AddGroup.class, DefaultGroup.class);

        taskProteinSequenceDesignService.save(dto);

        return new Result();
    }

    @PutMapping
    @ApiOperation("修改")
    @LogOperation("修改")
    @RequiresPermissions("medicine:taskproteinsequencedesign:update")
    public Result update(@RequestBody TaskProteinSequenceDesignDTO dto){
        //效验数据
        ValidatorUtils.validateEntity(dto, UpdateGroup.class, DefaultGroup.class);

        taskProteinSequenceDesignService.update(dto);

        return new Result();
    }

    @DeleteMapping
    @ApiOperation("删除")
    @LogOperation("删除")
    @RequiresPermissions("medicine:taskproteinsequencedesign:delete")
    public Result delete(@RequestBody Long[] ids){
        //效验数据
        AssertUtils.isArrayEmpty(ids, "id");

        taskProteinSequenceDesignService.delete(ids);

        return new Result();
    }

    @GetMapping("export")
    @ApiOperation("导出")
    @LogOperation("导出")
    @RequiresPermissions("medicine:taskproteinsequencedesign:export")
    public void export(@ApiIgnore @RequestParam Map<String, Object> params, HttpServletResponse response) throws Exception {
        List<TaskProteinSequenceDesignDTO> list = taskProteinSequenceDesignService.list(params);

        ExcelUtils.exportExcelToTarget(response, null, list, TaskProteinSequenceDesignExcel.class);
    }

}